import os.path as osp
from pathlib import Path

ROOT = Path(__file__).resolve().parents[0].parents[0]  # 项目根目录

# 数据集目录
DATASETS_DIR_PATH = osp.join(ROOT, "datasets")
# 输入目录
INPUT_DIR_PATH = osp.join(ROOT, "INPUT")
# 输出目录
OUTPUT_DIR_PATH = osp.join(ROOT, "OUTPUT")
# detector目录
DETECTOR_DIR_PATH = osp.join(ROOT, "Detector")
# DeepSORT目录
DEEPSORT_DIR_PATH = osp.join(ROOT, "DeepSORT")

DETECTOR = {
    "YAML_PATH": osp.join(DETECTOR_DIR_PATH, "yaml", "WiderPerson.yaml"),
    "ONNX_PATH": osp.join(DETECTOR_DIR_PATH, "onnx", "2", "best.onnx")
}

REID_MODEL = {
    "MODEL_NAME": "ResNet50",
    "WEIGHT_PATH": osp.join(DEEPSORT_DIR_PATH, "ReID", "runs", "weight", "best.pt")
}

# 测试图片
TEST_IMAGE_PATH = osp.join(INPUT_DIR_PATH, "bus.jpg")
# 测试视频
TEST_VIDEO_PATH = osp.join(INPUT_DIR_PATH, "football-video.mp4")
TEST_VIDEO2_PATH = osp.join(INPUT_DIR_PATH, "test.mkv")


if __name__ == '__main__':
    print(TEST_IMAGE_PATH)